#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Kronos Stock Prediction CLI

命令行接口模块
"""

import sys
import os

# 添加项目根目录到路径
current_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.dirname(current_dir)
sys.path.insert(0, project_root)

from examples.interactive_stock_prediction import InteractiveStockPredictor

def main():
    """主函数"""
    predictor = InteractiveStockPredictor()
    
    # 检查是否为测试模式
    if len(sys.argv) > 1 and sys.argv[1] == '--test':
        # 测试模式：使用示例股票代码
        test_codes = ['600036', '000001']  # 招商银行、平安银行
        predictor.run(test_mode=True, test_stock_codes=test_codes)
    elif len(sys.argv) > 1:
        # 命令行模式：直接指定股票代码
        stock_codes = sys.argv[1:]
        # 验证股票代码格式
        valid_codes = []
        for code in stock_codes:
            if code.isdigit() and len(code) == 6:
                valid_codes.append(code)
            else:
                print(f"⚠️  股票代码 {code} 格式不正确，已跳过")
        
        if valid_codes:
            print(f"📊 将预测以下股票: {', '.join(valid_codes)}")
            predictor.run(test_mode=True, test_stock_codes=valid_codes)
        else:
            print("❌ 没有有效的股票代码")
    else:
        # 正常交互模式
        predictor.run()

if __name__ == "__main__":
    main()
